package cn.ycc1.boot3websocket.config;

import cn.ycc1.boot3websocket.handler.DanmakuWebSocketHandler;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import org.springframework.web.socket.server.standard.ServerEndpointExporter;

/**
 * @author ycc
 * @date 2024/5/2
 */
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
//    @Bean
//    public ServerEndpointExporter serverEndpointExporter() {
//        return new ServerEndpointExporter();
//    }

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // 注册WebSocket处理器和拦截器
        registry.addHandler(new DanmakuWebSocketHandler(), "/websocket/danmaku")
                .setAllowedOrigins("*"); // 生产环境要设置具体的域名
    }
}
//开启WebSocket的支持，并把该类注入到spring容器中
//@Configuration
//public class WebSocketConfig {
//
//    @Bean
//    public ServerEndpointExporter serverEndpointExporter() {
//        return new ServerEndpointExporter();
//    }
//
//}
